/*
 * Copyright 2017 The Netty Project
 *
 * The Netty Project licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */
package io.netty.util.concurrent;

import io.netty.util.internal.ObjectUtil;

final class FastThreadLocalRunnable implements Runnable {

  private final Runnable runnable;

  private FastThreadLocalRunnable(Runnable runnable) {
    this.runnable = ObjectUtil.checkNotNull(runnable, "runnable");
  }

  @Override
  public void run() {
    try {
      runnable.run();
    } finally {
      FastThreadLocal.removeAll();
    }
  }

  static Runnable wrap(Runnable runnable) {
    return runnable instanceof FastThreadLocalRunnable ? runnable
        : new FastThreadLocalRunnable(runnable);
  }
}
